# 导入base模块
from common.base import Base, By


# TODO 定义LoginPage类，继承Base类
class LoginPage(Base):

    # 登陆页面路径
    login_url = 'http://172.16.140.55/ecshop/user.php'

    # TODO 定位账号输入框，输入数据
    def input_username(self, username):
        """
        调用Base里location_element方法
                attribute_element方法
        """
        self.find_element((By.NAME, 'username')).send_keys(username)

    # TODO 定位密码输入框，输入数据
    def input_password(self, password):
        """
        调用Base里location_element方法
        """
        self.find_element((By.NAME, 'password')).send_keys(password)

    # TODO 定位保存信息框，点击
    def remember(self):
        """
        调用Base里click方法
        """
        self.click((By.NAME, 'remember'))

    # TODO 定位登录按钮，点击
    def btn_login(self):
        self.click((By.NAME, 'submit'))

    # 定位到请登录按钮
    def please_login(self):
        self.click((By.LINK_TEXT, '请登录 '))

    # 定位登录文本
    def get_login_text(self):
        self.get_element_text((By.LINK_TEXT, '请登录 '))